package com.demo.common.utils;

import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import javax.imageio.ImageReadParam;
import javax.imageio.ImageReader;
import javax.imageio.stream.ImageInputStream;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.Iterator;

/**
 * 图片剪切工具类</p>
 *
 * @author liwenjun
 * @version 1.0.0
 * @date 2016/11/19 18:04
 */
public class ImageCut {


    /**
     * jpg图片格式
     */
    private static final String IMAGE_FORM_OF_JPG = "JPG";
    /**
     * png图片格式
     */
    private static final String IMAGE_FORM_OF_PNG = "PNG";


    /**
     * 返回包含所有当前已注册 ImageReader 的 Iterator，这些 ImageReader 声称能够解码指定格式。
     * 参数：formatName - 包含非正式格式名称 .（例如 "png" 或 "jpg"）等 。
     *
     * @param postFix 文件的后缀名
     * @return
     */
    public static Iterator<ImageReader> getImageReadersByFormatName(String postFix) {

        if (IMAGE_FORM_OF_JPG.equals(postFix)) {
            return ImageIO.getImageReadersByFormatName(IMAGE_FORM_OF_JPG);
        } else if (IMAGE_FORM_OF_JPG.equals(postFix)) {
            return ImageIO.getImageReadersByFormatName(IMAGE_FORM_OF_PNG);
        } else {
            return ImageIO.getImageReadersByFormatName(IMAGE_FORM_OF_JPG);
        }
    }

    /**
     * 获取inputFilePath的后缀名，如："e:/com.demo.modules.test.jpg"的后缀名为："jpg"<br>
     *
     * @param inputFilePath
     * @return
     */
    public static String getPostfix(String inputFilePath) {
        return inputFilePath.substring(inputFilePath.lastIndexOf(".") + 1).toUpperCase();
    }


    /**
     * 对图片裁剪，并把裁剪完蛋新图片保存 。
     *
     * @param multipartFile 源图片
     * @param folderPath    剪切图片存放路径
     * @param imgName       图片保存名称
     * @throws IOException
     */
    public static void cut(MultipartFile multipartFile, String folderPath, String imgName, int x, int y, int width, int height) throws IOException {
        ImageInputStream iis = null;
        try {
            // 读取图片文件
            //String imgPath = multipartFile.getOriginalFilename();

            // 获取文件的后缀名
            String postFix = getPostfix(imgName);
            System.out.println("图片格式为：" + postFix);
            /*
             * 返回包含所有当前已注册 ImageReader 的 Iterator，这些 ImageReader 声称能够解码指定格式。
             * 参数：formatName - 包含非正式格式名称 .（例如 "jpeg" 或 "tiff"）等 。
             */
            Iterator<ImageReader> it = getImageReadersByFormatName(postFix);

            ImageReader reader = it.next();
            // 获取图片流
            iis = ImageIO.createImageInputStream(multipartFile.getInputStream());

            /*
             * <p>iis:读取源.true:只向前搜索 </p>.将它标记为 ‘只向前搜索'。
             * 此设置意味着包含在输入源中的图像将只按顺序读取，可能允许 reader 避免缓存包含与以前已经读取的图像关联的数据的那些输入部分。
             */
            reader.setInput(iis, true);

            /*
             * <p>描述如何对流进行解码的类<p>.用于指定如何在输入时从 Java Image I/O
             * 框架的上下文中的流转换一幅图像或一组图像。用于特定图像格式的插件 将从其 ImageReader 实现的
             * getDefaultReadParam 方法中返回 ImageReadParam 的实例。
             */
            ImageReadParam param = reader.getDefaultReadParam();

            /*
             * 图片裁剪区域。Rectangle 指定了坐标空间中的一个区域，通过 Rectangle 对象
             * 的左上顶点的坐标（x，y）、宽度和高度可以定义这个区域。
             */
            Rectangle rect = new Rectangle(x, y, width, height);

            // 提供一个 BufferedImage，将其用作解码像素数据的目标。
            param.setSourceRegion(rect);
            /*
             * 使用所提供的 ImageReadParam 读取通过索引 imageIndex 指定的对象，并将 它作为一个完整的
             * BufferedImage 返回。
             */
            BufferedImage bi = reader.read(0, param);

            //文件保存路径是否存在
            File serFile = new File(folderPath);
            if (!serFile.exists()) {
                serFile.mkdirs();
            }

            File imgFile = new File(folderPath + imgName);

            // 保存新图片
            ImageIO.write(bi, postFix, imgFile);
        } finally {
            if (iis != null) {
                iis.close();
            }
        }

    }

}
